Introduction to Data Science in Python coursera assignment 2